package Leetcode.动态规划;

import java.util.Arrays;

/**
 * @Author: kirito
 * @Date: 2024/4/4 14:21
 * @Description:
 *买卖股票的最佳时机 IV
 * 给你一个整数数组 prices 和一个整数 k ，其中 prices[i] 是某支给定的股票在第 i 天的价格。
 *
 * 设计一个算法来计算你所能获取的最大利润。你最多可以完成 k 笔交易。也就是说，你最多可以买 k 次，卖 k 次。
 *
 * 注意：你不能同时参与多笔交易（你必须在再次购买前出售掉之前的股票）。
 *
 *
 *
 * 示例 1：
 *
 * 输入：k = 2, prices = [2,4,1]
 * 输出：2
 * 解释：在第 1 天 (股票价格 = 2) 的时候买入，在第 2 天 (股票价格 = 4) 的时候卖出，这笔交易所能获得利润 = 4-2 = 2 。
 * 示例 2：
 *
 * 输入：k = 2, prices = [3,2,6,5,0,3]
 * 输出：7
 * 解释：在第 2 天 (股票价格 = 2) 的时候买入，在第 3 天 (股票价格 = 6) 的时候卖出, 这笔交易所能获得利润 = 6-2 = 4 。
 *      随后，在第 5 天 (股票价格 = 0) 的时候买入，在第 6 天 (股票价格 = 3) 的时候卖出, 这笔交易所能获得利润 = 3-0 = 3 。
 *
 *
 * 提示：
 *
 * 1 <= k <= 100
 * 1 <= prices.length <= 1000
 * 0 <= prices[i] <= 1000
 */

public class maxProfit4 {
    /**
     * 愈加离谱，不会不会。。。
     * @param k
     * @param prices
     * @return
     */
    public int maxProfit(int k, int[] prices) {
        if (prices.length == 0) {
            return 0; // 如果价格数组为空，直接返回0，因为没有交易可做
        }

        int n = prices.length;
        k = Math.min(k, n / 2); // 限制交易次数不能超过n/2，因为超过这个次数就等同于不限交易次数
        int[][] buy = new int[n][k + 1]; // 初始化买入利润数组
        int[][] sell = new int[n][k + 1]; // 初始化卖出利润数组

        buy[0][0] = -prices[0]; // 第一天的第一次买入利润为负的第一天价格
        sell[0][0] = 0; // 第一天的第一次卖出利润为0，因为没有买入
        for (int i = 1; i <= k; ++i) {
            buy[0][i] = sell[0][i] = Integer.MIN_VALUE / 2; // 初始化第一天的其他交易为不可能的情况
        }

        for (int i = 1; i < n; ++i) {
            buy[i][0] = Math.max(buy[i - 1][0], sell[i - 1][0] - prices[i]); // 计算第i天的第一次买入利润
            for (int j = 1; j <= k; ++j) {
                buy[i][j] = Math.max(buy[i - 1][j], sell[i - 1][j] - prices[i]); // 计算第i天的第j次买入利润
                sell[i][j] = Math.max(sell[i - 1][j], buy[i - 1][j - 1] + prices[i]); // 计算第i天的第j次卖出利润
            }
        }

        return Arrays.stream(sell[n - 1]).max().getAsInt(); // 返回最后一天所有可能交易中的最大卖出利润
    }

}
